package month_1;

import java.util.ArrayList;
import java.util.List;

/**
 * @author： pang-yun
 * @date： 2022-01-07 11:40
 * 功能目的 ：用于  列表扁平化
 */

public class Flatten_2 {
    public static void main(String[] args) {
        ArrayList<Object> list = new ArrayList<>();
        ArrayList<Object> l = new ArrayList<>();
        list.add(1);
        l.add(2);
        l.add(22);
        l.add(24);
        l.add(25);
        list.add(l);
        List<Integer> flatten = flatten(list);
        flatten.stream().forEach(System.out::println);
    }

    public static List<Integer> flatten(List<Object> list) {
        ArrayList<Integer> result = new ArrayList<>();
        list.stream().forEach(element -> {
            if (element instanceof Integer) {
                result.add((Integer) element);
            } else {
                result.addAll((List)element);
            }
        });
        return result;
    }

}
